import pandas as pd
import os

data_list = {}
p_path = os.getcwd()
dir_path = p_path + "/数据源/TEMU结算数据/"

for fname in os.listdir(dir_path):
    if fname.endswith(".xlsx"):
        df_excel = pd.read_excel(dir_path + fname, sheet_name=None)
        store_name = fname.split('-')[0]
        print(f'店铺名称：{store_name}')
        for sheet_name, df_data in df_excel.items():
            df_data.insert(0, column='店铺', value=store_name)

            if sheet_name in data_list.keys():
                data_list[sheet_name] = pd.concat([data_list[sheet_name], df_data], ignore_index=True)
                print('---------------')
                print('已经存在了，合并数据')
            else:
                data_list[sheet_name] = df_data

            print(f'表名：{sheet_name},行数：{df_data.count()}')

for sheet_name, data in data_list.items():
    data.to_excel(f"{p_path}/处理过的数据/1.{sheet_name}.xlsx", index=False)

    print(f'合并 {sheet_name} 数据完成，总数据量：{len(data)}条')

